python Pandas : Add column to grouped DataFrame with method chaining